Skip to content

gpu, generalise, etc. some changes#1

Open
prahlad-ctrl wants to merge 1 commit intoavbiswas:mainfrom
prahlad-ctrl:main
Open

gpu, generalise, etc. some changes#1
prahlad-ctrl wants to merge 1 commit intoavbiswas:mainfrom
prahlad-ctrl:main

Conversation

@prahlad-ctrl
Copy link

  • centralized all hyperparameters and paths into the new 'config.yaml' file to eliminate hardcoded values
  • optimized performance by moving image encoding to the GPU and refactoring dataloaders to use a shared base class
  • enhanced the training scripts with config-driven settings, improved logging, and standardized checkpoint management.
  • consolidated evaluation into a unified inference script and removed redundant demo files

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant